// Copyright (C) 2014-2020  CEA/DEN, EDF R&D
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
//
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
//
// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
//

#include "CollectionPlugin_GroupShape.h"
#include "ModelAPI_ResultBody.h"
#include <TopoDS.hxx>
#include <TopoDS_Iterator.hxx>

void CollectionPlugin_GroupShape::execute()
{
  ResultGroupPtr aGroup;
  CollectionPlugin_GroupMerge::execute(aGroup);
  GeomShapePtr aCompound = aGroup->shape();

  const TopoDS_Shape & aShape = aCompound->impl<TopoDS_Shape>();

  if(aShape.NbChildren() == 1)
  {
    /// unique shape, remove compound on type

    TopoDS_Iterator anIt(aShape);
    TopoDS_Shape aShape = anIt.Value();
    std::shared_ptr<GeomAPI_Shape> aRes(new GeomAPI_Shape);
    aRes->setImpl(new TopoDS_Shape(aShape));
    aGroup->store(GeomShapePtr());
    aGroup->store(aRes);
    aCompound = aGroup->shape();
  }

  std::shared_ptr<ModelAPI_ResultBody> theResultBody = document()->createBody(data());
  theResultBody->store(GeomShapePtr());
  if(!aCompound->empty())
    theResultBody->store(aCompound);
  setResult(theResultBody);
}
